/*
给定整数数组 A，每次 move 操作将会选择任意 A[i]，并将其递增 1。

返回使 A 中的每个值都是唯一的最少操作次数。

示例 1:

输入：[1,2,2]
输出：1
解释：经过一次 move 操作，数组将变为 [1, 2, 3]。
示例 2:

输入：[3,2,1,2,1,7]
输出：6
解释：经过 6 次 move 操作，数组将变为 [3, 4, 1, 2, 5, 7]。
可以看出 5 次或 5 次以下的 move 操作是不能让数组的每个值唯一的。
提示：

0 <= A.length <= 40000
0 <= A[i] < 40000

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/minimum-increment-to-make-array-unique
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
*/

// town 2020.8.13 复杂度较高 未通过
class Solution {
public:
    int minIncrementForUnique(vector<int>& A) {
        
        map<int,int> umap;
        for(int i=0;i<A.size();i++){
            umap[A[i]]++;
        }
        for(auto& iter:umap){
            cout<<iter.first<<" "<<iter.second<<endl;
        }
        int res=0;
        int empty=0;
        auto iter=umap.begin();
        while(iter!=umap.end()){
            empty=iter->first; 
            while(iter->second>1){
                //cout<<iter->first<<" "<<iter->second<<endl;
                while(umap.count(empty)!=0){
                    empty++;
                }
                //cout<<"empty:"<<empty<<endl;
                umap[empty]=1;
                res=res+empty-iter->first;
                iter->second --;
            }
            ++iter;
        }
        for(auto& iter:umap){
            cout<<iter.first<<" "<<iter.second<<endl;
        }
        return res;
        
    }
};

// town 2020.8.13 O(N+K)
class Solution {
public:
    int minIncrementForUnique(vector<int>& A) {
        int dict[40000]= {0};
        int max_num=0;
        int res=0;
        for(auto& i:A){
            dict[i]++;
            max_num=max(max_num,i);
        }
        for(int i=0;i<max_num;i++){
            if(dict[i]>1){
                dict[i+1]+=dict[i]-1;
                res+=dict[i]-1;
            }
        }
        if(dict[max_num]>1){
            res+=(dict[max_num]-1)*dict[max_num]/2;
        }
        return res;
    }
};

// town 2020.8.13 O(nlogn)
class Solution {
public:
    int minIncrementForUnique(vector<int>& A) {
        sort(A.begin(),A.end());
        int result=0;
        for(int i=1;i<A.size();i++){
            if(A[i]<=A[i-1]){
                int tmp=A[i];
                A[i]=A[i-1]+1;
                result+=A[i]-tmp;
            }
        }
        return result;
    }
};